package com.gyg.bro.utils;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.OSSObject;
import com.aliyun.oss.model.OSSObjectSummary;
import com.aliyun.oss.model.ObjectListing;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.FilenameUtils;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletResponse;
import java.io.BufferedInputStream;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.util.Date;
import java.util.List;
import java.util.UUID;
import java.util.stream.Collectors;

@Component
@Slf4j
public class OSSUtil {

    @Value("${aliyun.oss.endpoint}")
    private String endPoint;

    @Value("${aliyun.oss.accessKeyId}")
    private String accessKeyId;

    @Value("${aliyun.oss.accessSecret}")
    private String accessKeySecret;

    @Value("${aliyun.oss.bucketName}")
    private String bucketName;

    /**
     * 文件上传
     * @param objectName
     * @param bytes
     */
    public void upload(String objectName, byte[] bytes) {
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endPoint, accessKeyId, accessKeySecret);
        // 填写Bucket名称、Object完整路径和本地文件的完整路径。Object完整路径中不能包含Bucket名称。
        // 如果未指定本地路径，则默认从示例程序所属项目对应本地路径中上传文件。
        // 上传文件。
        ossClient.putObject(bucketName, objectName, new ByteArrayInputStream(bytes));
        log.info(objectName + "文件上传成功");
        // 关闭OSSClient。
        ossClient.shutdown();
    }

    /**
     * 文件下载
     * @param response
     * @param objectName
     * @param filename 原始文件名
     */
    public void download(HttpServletResponse response, String objectName, String filename) {
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endPoint, accessKeyId, accessKeySecret);
        // ossObject包含文件所在的存储空间名称、文件名称、文件元信息以及一个输入流。
        OSSObject ossObject = ossClient.getObject(bucketName, objectName);
        // 读取文件内容。
        BufferedInputStream input = new BufferedInputStream(ossObject.getObjectContent());
        try {
            //通知浏览器以附件形式下载
            response.setHeader("Content-Disposition","attachment;filename="+ URLEncoder.encode(filename,"utf-8"));
            byte[] buffBytes = new byte[1024];
            OutputStream outputStream = response.getOutputStream();
            int read = 0;
            while ((read = input.read(buffBytes)) != -1) {
                outputStream.write(buffBytes, 0, read);
            }
            // 数据读取完成后，获取的流必须关闭，否则会造成连接泄漏，导致请求无连接可用，程序无法正常工作。
            outputStream.flush();
            outputStream.close();
            input.close();
            log.info(objectName + "文件下载成功");
        } catch (IOException e) {
            log.error(objectName + "文件下载失败");
        }
        // 关闭OSSClient。
        ossClient.shutdown();
    }

    /**
     * 生成随机文件名
     * @param objectName
     * @param fileName
     * @return
     */
    public String getFileName(String objectName, String fileName) {
        String extension = FilenameUtils.getExtension(fileName);
        String date = TimeUtilCustomize.parseDateToStr(TimeUtilCustomize.DATE_DAY_FORMAT, new Date());
        String uuid = UUID.randomUUID().toString();
        return objectName + date + "-" + uuid + "." + extension;
    }

    /**
     * 获取oss文件访问url
     * @param objectName
     * @return
     */
    public String getUrl(String objectName) {
        return "http://gaoyiguan.cn/" + objectName;
    }

    public List<String> list(String keyPrefix) {
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endPoint, accessKeyId, accessKeySecret);
        // 列举文件。如果不设置KeyPrefix，则列举存储空间下的所有文件。如果设置KeyPrefix，则列举包含指定前缀的文件。
        ObjectListing objectListing = ossClient.listObjects(bucketName, keyPrefix);
        List<OSSObjectSummary> sums = objectListing.getObjectSummaries();
        if (sums.isEmpty()) {
            return null;
        }

        return sums.stream()
                .map(OSSObjectSummary::getKey)
                .collect(Collectors.toList());
    }

    /**
     * 删除文件
     * @param objectName
     */
    public void delete(String objectName) {
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endPoint, accessKeyId, accessKeySecret);
        // 删除文件。如需删除文件夹，请将ObjectName设置为对应的文件夹名称。如果文件夹非空，则需要将文件夹下的所有object删除后才能删除该文件夹。
        ossClient.deleteObject(bucketName, objectName);
        log.info(objectName + "文件删除成功！");
        // 关闭OSSClient。
        ossClient.shutdown();
    }

    /**
     * 批量删除文件
     * @param objectNameList
     */
    public void deleteAll(List<String> objectNameList) {
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endPoint, accessKeyId, accessKeySecret);
        // 删除文件。如需删除文件夹，请将ObjectName设置为对应的文件夹名称。如果文件夹非空，则需要将文件夹下的所有object删除后才能删除该文件夹。
        for (String objectName : objectNameList) {
            ossClient.deleteObject(bucketName, objectName);
            log.info(objectName + "文件删除成功！");
        }
        // 关闭OSSClient。
        ossClient.shutdown();
    }

}
